package com.alibaba.csp.sentinel.demo;

import com.alibaba.csp.sentinel.Entry;
import com.alibaba.csp.sentinel.SphU;
import com.alibaba.csp.sentinel.slots.block.BlockException;
import com.alibaba.csp.sentinel.slots.block.RuleConstant;
import com.alibaba.csp.sentinel.slots.block.flow.FlowRule;
import com.alibaba.csp.sentinel.slots.block.flow.FlowRuleManager;

import java.util.ArrayList;
import java.util.List;

/**
 * 参考文档链接：https://xiaolyuh.blog.csdn.net/article/details/107998973
 */
public class SentinelMain {

    /**
     * 先是创建entry对象。需要保护的方法放在SphU.entry("自定义资源名")、 entry.exit()之间包起来。
     * 超过限制时抛出BlockException 异常，进行业务处理。可见entry帮我们做了指标收集、超过阀值的判断等操作。
     * 让我们不需要感知。不同的业务都需要写重复代码。
     * @param args
     */
    public static void main(String[] args) {
        // 配置规则.
        initFlowRules();

        while (true) {
            Entry entry = null;
            // 1.5.0 版本开始可以直接利用 try-with-resources 特性
            try  {
                // 资源名可使用任意有业务语义的字符串
                entry = SphU.entry("HelloWorld");
                // 被保护的逻辑
                System.out.println("hello world");
            } catch (BlockException ex) {
                // 处理被流控的逻辑
                // 资源访问阻止，被限流或被降级
                System.out.println("blocked!");
            }finally {
                if (entry != null) {
                    entry.exit();
                }
            }
        }
    }

    private static void initFlowRules(){
        List<FlowRule> rules = new ArrayList<>();
        FlowRule rule = new FlowRule();
        rule.setResource("HelloWorld");
        rule.setGrade(RuleConstant.FLOW_GRADE_QPS);
        // Set limit QPS to 20.
        rule.setCount(20);
        rules.add(rule);
        FlowRuleManager.loadRules(rules);
    }
}
